17905f56c9c697dd3b0c34cfbc184adb3d643af8,modules/repository-datastax/src/test/java/com/intuit/wasabi/repository/cassandra/impl/CassandraAssignmentsRepositoryTest.java,CassandraAssignmentsRepositoryTest,testGetAssignmentsMultiple,#,245

Before Change


                Experiment.withID(Experiment.ID.valueOf(random)).build()
        );
        for (ExperimentUserByUserIdContextAppNameExperimentId item : mocked) {
            experimentTable.put(Experiment.ID.valueOf(item.getExperimentId()),
                    Experiment.Label.valueOf("test-" + item.getBucket()),
                    Experiment.withID(Experiment.ID.valueOf(item.getExperimentId()))
                            .withLabel(Experiment.Label.valueOf("test-" + item.getBucket())).build()
            );
        }

        doReturn(mocked.stream()).when(spyRepository).getUserIndexStream(anyString(), anyString(), anyString());

After Change


    @Test
    public void testGetAssignmentsMultiple() {
        Experiment.ID expId1 = Experiment.ID.newInstance();
        Experiment.ID expId2 = Experiment.ID.newInstance();
        Date endTime = new Date(System.currentTimeMillis() + 30 * 24 * 60 * 60 * 1000);

        Experiment exp1 = Experiment.withID(expId1).withEndTime(endTime)
                .withLabel(Experiment.Label.valueOf("Exp1")).build();
        Experiment exp2 = Experiment.withID(expId2).withEndTime(endTime)
                .withLabel(Experiment.Label.valueOf("Exp2")).build();

        List<ExperimentUserByUserIdContextAppNameExperimentId> mocked = new ArrayList<>();
        mocked.add(ExperimentUserByUserIdContextAppNameExperimentId.builder()
                .appName(APPLICATION_NAME.toString())
                .experimentId(expId1.getRawID())
                .context("test")
                .bucket("bucket1")
                .build()
        );
        mocked.add(ExperimentUserByUserIdContextAppNameExperimentId.builder()
                .appName(APPLICATION_NAME.toString())
                .experimentId(expId2.getRawID())
                .context("test")
                .bucket("bucket2")
                .build()
        );

        Map<Experiment.ID, Experiment> experimentMap = newHashMap();
        experimentMap.put(expId1, exp1);
        experimentMap.put(expId2, exp2);

        doReturn(mocked.stream()).when(spyRepository).getUserIndexStream(anyString(), anyString(), anyString());